minor fix before merging