Correct import for test and add debug mode