diff --git a/lib/get_answer.py b/lib/get_answer.py index a5eb81c..6a7d8e2 100644 --- a/lib/get_answer.py +++ b/lib/get_answer.py @@ -95,7 +95,7 @@ def _isdir(topic): return os.path.isdir(topic) def _get_answers_and_dirs(): topics = glob.glob(PATH_CHEAT_SHEETS + "*") - answer_dirs = [_remove_initial_underscore(os.path.split(topic)[1]) + answer_dirs = [_remove_initial_underscore(os.path.split(topic)[1]).rstrip('/')+'/' for topic in topics if _isdir(topic)] answers = [os.path.split(topic)[1] for topic in topics if not _isdir(topic)] return answer_dirs, answers