Added tests for decorator