Fixed the bug with test taking