fixed a small problem with arguments