Minor refactor

Fix path error