Add import for marked to tests