minor fix, working