Fixed missing import that broke category as course calculation