now works with the new net layer