fixed a small problem, still wrong