Removed wrong lines, fixed