correct the main.cpp collsion