Forgot to save before push. Refactor