minor fix, got it working