minor fix, but everything is working