Fixed some module mismatch between test and application