Fix last commit, minor issue