Small fix to make the project runnable again