minor fix on assignment