assignment - small error bugfixing