Fix accidentally commited lines in previous commit