Removed deprecated annotation to avoid test error
Added tests