Updated source locations to test and main, maybe it could be improved
to: src/main src/test